# +----------------------------------------------------------------------
# | ChatWork智能聊天办公系统
# +----------------------------------------------------------------------
# | 软件声明: 本系统并非自由软件,未经授权任何形式的商业使用均属非法。
# | 版权保护: 任何企业和个人不允许对程序代码以任何形式任何目的复制/分发。
# | 授权要求: 如有商业使用需求,请务必先与版权所有者取得联系并获得正式授权。
# +----------------------------------------------------------------------
# | Author: ChatWork Team <2474369941@qq.com>
# +----------------------------------------------------------------------
import json
import time
from hypertext import PagingResult
from common.utils.urls import UrlUtil
from common.utils.tools import ToolsUtil
from common.enums.models import ModelsEnum
from common.models.docs import DocsPromptModel
from apps.admin.schemas.apply.docs import prompt_schema as schema


class DocsPromptService:

    @classmethod
    async def lists(cls, params: schema.DocsPromptSearchIn) -> PagingResult[schema.DocsPromptListVo]:
        """
        模式列表。

        Args:
            params (schema.DocsPromptSearchIn): 查询参数。

        Returns:
            PagingResult[schema.DocsPromptListVo]: 分页列表Vo。

        Author:
            zero
        """
        # 查询条件
        where = DocsPromptModel.build_search({
            "%like%": ["name"]
        }, params.__dict__)

        # 查询数据
        _model = DocsPromptModel.filter(*where).order_by("-sort", "-id")
        _pager = await DocsPromptModel.paginate(
            model=_model,
            page_no=params.page_no,
            page_size=params.page_size,
            fields=["id", "code", "icon", "name", "intro", "sort", "is_disable", "is_builtin", "create_time"]
        )

        # 处理数据
        _lists = []
        for item in _pager.lists:
            _lists.append(schema.DocsPromptListVo(
                id=item["id"],
                code=item["code"],
                icon=await UrlUtil.to_absolute_url(item["icon"]),
                name=item["name"],
                intro=item["intro"],
                sort=item["sort"],
                is_builtin=item["is_builtin"],
                is_disable=item["is_disable"],
                create_time=item["create_time"]
            ))

        _pager.lists = _lists
        return _pager

    @classmethod
    async def detail(cls, id_: int) -> schema.DocsPromptDetailVo:
        """
        模式详情。

        Args:
            id_ (int): 查询参数。

        Returns:
            PagingResult[schema.DocsPromptDetailVo]: 分页列表Vo。

        Author:
            zero
        """
        detail = await DocsPromptModel.filter(id=id_).get()
        return schema.DocsPromptDetailVo(
            id=detail.id,
            code=detail.code,
            icon=await UrlUtil.to_absolute_url(detail.icon),
            name=detail.name,
            intro=detail.intro,
            sort=detail.sort,
            is_disable=detail.is_disable,
            abstract_conf=json.loads(detail.abstract_conf or "[]"),
            dialogue_conf=json.loads(detail.dialogue_conf or "{}"),
        )

    @classmethod
    async def add(cls, post: schema.DocsPromptAddIn):
        """
        模式添加。

        Args:
            post (schema.DocsPromptAddIn): 添加参数。

        Author:
            zero
        """
        code: str = await ToolsUtil.make_rand_sn(DocsPromptModel, "code", 8)
        await DocsPromptModel.create(
            code=code,
            icon=UrlUtil.to_relative_url(post.icon),
            name=post.name,
            intro=post.intro,
            sort=post.sort,
            is_disable=post.is_disable,
            create_time=int(time.time()),
            update_time=int(time.time()),
            abstract_conf=json.dumps(post.abstract_conf, ensure_ascii=False),
            dialogue_conf=json.dumps(post.dialogue_conf, ensure_ascii=False)
        )

    @classmethod
    async def edit(cls, post: schema.DocsPromptEditIn):
        """
        模式编辑。

        Args:
            post (schema.DocsPromptEditIn): 编辑参数。

        Author:
            zero
        """
        await DocsPromptModel.filter(id=post.id).update(
            icon=UrlUtil.to_relative_url(post.icon),
            name=post.name,
            intro=post.intro,
            sort=post.sort,
            is_disable=post.is_disable,
            update_time=int(time.time()),
            abstract_conf=json.dumps(post.abstract_conf, ensure_ascii=False),
            dialogue_conf=json.dumps(post.dialogue_conf, ensure_ascii=False)
        )

    @classmethod
    async def delete(cls, id_: int):
        """
        模式删除。

        Args:
            id_ (int): 模式id。

        Author:
            zero
        """
        await DocsPromptModel.filter(id=id_, is_delete=0).get()
        await DocsPromptModel.filter(id=id_).update(
            is_delete=1,
            delete_time=int(time.time())
        )
